package xyz.bernkastel.goods_in_great_demand.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.util.DigestUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import xyz.bernkastel.goods_in_great_demand.common.CheckCodeUtil;
import xyz.bernkastel.goods_in_great_demand.common.R;
import xyz.bernkastel.goods_in_great_demand.dto.EmployeeDto;
import xyz.bernkastel.goods_in_great_demand.entity.Employee;
import xyz.bernkastel.goods_in_great_demand.service.EmployeeService;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.time.LocalDateTime;
import java.util.Locale;

@Slf4j
@RestController
@RequestMapping("/employee")
public class EmployeeController {
    @Value("${goods.path}")
    private String basePath;    //存储验证码图片的路径

    @Autowired
    private EmployeeService employeeService;

    /**
     * 员工登录
     *
     * @param request  返回的数据体
     * @param employee 收到的结构体，json格式
     * @return
     */
    @PostMapping("/login")
    public R<Employee> login(HttpServletRequest request, @RequestBody EmployeeDto employee) {
        /*
        功能分析
        1、将页面提交的密码password进行md5加密处理
        2、根据页面提交的用户名username查询数据库
        3、如果没有查询到则返回登录失败结果
        4、密码比对，如果不一致则返回登录失败结果
        5、查看员工状态，如果为已禁用状态，则返回员工已禁用结果
        6、登录成功，将员工id存入Session并返回登录成功结果
         */
        String trueCheckCode = (String) request.getSession().getAttribute("checkcode");
        trueCheckCode = trueCheckCode.toLowerCase();
        String inputCheckCode = employee.getCheckcode().toLowerCase();

        if (!trueCheckCode.equals(inputCheckCode)) {
            return R.error("验证码错误");
        }
        //1、将页面提交的密码password进行md5加密处理
        String password = employee.getPassword();//得到密码
        password = DigestUtils.md5DigestAsHex(password.getBytes());//得到MD5加密后的字符串

        //2、根据页面提交的用户名username查询数据库
        LambdaQueryWrapper<Employee> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Employee::getUsername, employee.getUsername());//比对数据库
        Employee emp = employeeService.getOne(queryWrapper);//因为Username字段在数据库中存在唯一约束，所以得到一个即可
        //3、如果没有查询到则返回登录失败结果
        if (emp == null) {
            return R.error("登陆失败");
        }

        //4、密码比对，如果不一致则返回登录失败结果
        if (!emp.getPassword().equals(password)) {
            return R.error("密码错误");
        }
        //5、查看员工状态，如果为已禁用状态，则返回员工已禁用结果
        if (emp.getStatus() == 0) {
            return R.error("账号已禁用");
        }
        //6、登录成功，将员工id存入Session并返回登录成功结果
        request.getSession().setAttribute("employee", emp.getId());


        return R.success(emp);
    }

    /**
     * 员工退出
     *
     * @param request
     * @return
     */
    @PostMapping("/logout")
    public R<String> logout(HttpServletRequest request) {
        //清理Session中的信息
        request.getSession().removeAttribute("employee");
        return R.success("退出成功");
    }

    /**
     * 新增员工
     *
     * @param request
     * @param employee
     * @return
     */
    @PostMapping
    public R<String> save(HttpServletRequest request, @RequestBody Employee employee) {
        log.info("新增员工，员工信息：{}", employee.toString());

        //设置初始密码123456，后期员工登录后自行修改
        employee.setPassword(DigestUtils.md5DigestAsHex("123456".getBytes()));
        //使用自动填充完成如下功能
//        employee.setCreateTime(LocalDateTime.now());
//        employee.setUpdateTime(LocalDateTime.now());
//
//        //获得当前登录用户id
//        Long employeeId = (Long) request.getSession().getAttribute("employee");
//        employee.setCreateUser(employeeId);
//        employee.setUpdateUser(employeeId);

        employeeService.save(employee);
        return R.success("新增员工成功");
    }

    /**
     * 分页
     *
     * @param page     第几页
     * @param pageSize 每页的显示数量
     * @param name     指定的用户名
     * @return
     */
    @GetMapping("/page")
    public R<Page> page(int page, int pageSize, String name) {
        //分页构造器
        Page pageInfo = new Page(page, pageSize);
        //条件构造器
        LambdaQueryWrapper<Employee> lambdaQueryWrapper = new LambdaQueryWrapper();
        //添加过滤条件
        lambdaQueryWrapper.like(!StringUtils.isEmpty(name), Employee::getName, name);//对应sql语句的like查询
        //添加排序条件
        lambdaQueryWrapper.orderByDesc(Employee::getUpdateTime);//按照更新时间排序
        //执行查询
        employeeService.page(pageInfo, lambdaQueryWrapper);//不需要返回值，函数内部自动封装

        return R.success(pageInfo);
    }

    /**
     * 根据id修改员工信息
     *
     * @param employee
     * @return
     */
    @PutMapping
    public R<String> update(HttpServletRequest request, @RequestBody Employee employee) {
        log.info(employee.toString());

        //JS对Long数据处理会丢失精度，需要转换成String
        Long employeeId = (Long) request.getSession().getAttribute("employee");

        employee.setUpdateTime(LocalDateTime.now());
        employee.setUpdateUser(employeeId);
        employeeService.updateById(employee);
        return R.success("更新用户信息成功");
    }

    /**
     * 根据id来查询员工信息
     *
     * @param id
     * @return
     */
    @GetMapping("/{id}")
    public R<Employee> getById(@PathVariable Long id) {
        log.info("根据id查询员工信息");
        Employee employee = employeeService.getById(id);
        if (employee != null) {
            return R.success(employee);
        }
        return R.error("未查询到对应员工信息");
    }

    @GetMapping("/checkcode")
    public R<String> getCheckCode(HttpServletRequest request) throws IOException {
        log.info("获取验证码");
        String token = Long.toString(System.currentTimeMillis());//取现在时间创建时间戳
        OutputStream fos = new FileOutputStream(basePath + token + ".jpg");
        String checkcode = CheckCodeUtil.outputVerifyImage(100, 50, fos, 4);
        request.getSession().setAttribute("checkcode", checkcode);
        return R.success(token + ".jpg");
    }
}
